package cn.yunwei.module.service;

import cn.yunwei.module.config.MinioUtil;
import cn.yunwei.module.dal.dataobject.ModulePerfileDO;
import cn.yunwei.module.dal.mysql.moduleperfile.ModulePerFileMapper;
import com.xxl.job.core.handler.annotation.XxlJob;
import jakarta.annotation.Resource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.time.LocalDateTime;

/**
 * @author wangbinggui
 * @date 2024/9/9
 */
@Service
public class ModulePerFileServierImpl implements ModulePerFileServier {

    @Resource
    private ModulePerFileMapper modulePerFileMapper;

    @Resource
    private MinioUtil minioUtil;
    private static final Logger log = LoggerFactory.getLogger(ModulePerFileServierImpl.class);

    //将地址放在配置文件里面配置
    @Value("${file.source-folder}")
    private String sourceFolderPath;
    //存储在minio上的文件夹，放在配置文件里面配置
    @Value("${minioFile.object-prefix}")
    private String objectPrefix;

    @Override
    public ModulePerfileDO getFileInfoByDataId(Integer dataId) {
        return null;
    }

    /**
     * 实现文件自动上传到minio上面去
     */
    @Override
//    @Scheduled(fixedDelay = 3000)
    @XxlJob("uploadFile")
    public void uploadFile() {
        log.info("定时任务执行时间: {}", LocalDateTime.now());

        java.io.File sourceFolder = new java.io.File(sourceFolderPath);
        java.io.File[] files = sourceFolder.listFiles();

        if (files != null) {
            for (java.io.File file : files) {
                try (InputStream in = new FileInputStream(file)) {
                    // 构造目标文件路径
                    String objectName = objectPrefix + file.getName();
                    String fileHash = calculateFileHash(in);

                    ModulePerfileDO existingFile = modulePerFileMapper.getFileInfoByFileName(file.getName());

                    if (existingFile == null) {
                        // 检查文件夹路径是否存在
                        if (!minioUtil.doesFolderExist(objectPrefix)) {
                            // 如果不存在，创建文件夹路径（对象存储中只是模拟路径）
                            minioUtil.createFolder(objectPrefix);
                            log.info("已创建远程文件夹: {}", objectPrefix);
                        }

                        // 上传文件
                        try (InputStream fileInputStream = new FileInputStream(file)) {
                            minioUtil.uploadFile(objectName, fileInputStream, "application/octet-stream");
                        }

                        // 保存文件信息到数据库
                        ModulePerfileDO uploadedFile = new ModulePerfileDO();
                        uploadedFile.setFileName(file.getName());
                        uploadedFile.setFileHash(fileHash);

                        // 去掉文件名的后缀
                        String fileNameParts = file.getName();
                        if (fileNameParts.lastIndexOf(".") != -1) {
                            fileNameParts = fileNameParts.substring(0, fileNameParts.lastIndexOf("."));
                        }
                        uploadedFile.setBatchNumber(fileNameParts);
                        modulePerFileMapper.insert(uploadedFile);

                        log.info("文件上传成功: {}", file.getName());
                        // 删除本地文件
                        deleteFile(file);

                    } else {
                        log.info("文件已上传，跳过: {}", file.getName());
                    }
                } catch (Exception e) {
                    log.error("文件上传时出错: {}", e.getMessage(), e);
                }
            }
        } else {
            log.info("文件夹中没有找到文件。");
        }
    }

    // 计算文件的MD5哈希值
    private String calculateFileHash(InputStream inputStream) throws NoSuchAlgorithmException, IOException {
        MessageDigest md = MessageDigest.getInstance("MD5");
        byte[] buffer = new byte[8192];
        int read;
        while ((read = inputStream.read(buffer)) != -1) {
            md.update(buffer, 0, read);
        }
        byte[] hash = md.digest();

        // 将字节数组转换为十六进制字符串
        StringBuilder hexHash = new StringBuilder();
        for (byte b : hash) {
            hexHash.append(String.format("%02x", b));
        }

        return hexHash.toString();
    }

    /**
     * 删除本地文件
     */
    private void deleteFile(java.io.File file) {
        // 尝试删除文件，并检查是否成功
        if (file.exists()) {
            boolean isDeleted = file.delete();
            if (isDeleted) {
                log.info("本地文件已删除: {}", file.getName());
            } else {
                log.warn("无法删除本地文件: {}", file.getName());

                // 文件删除失败时，尝试使用垃圾回收机制解决文件占用问题
                System.gc();  // 请求 JVM 执行垃圾回收，释放可能锁定的资源
                try {
                    // 短暂等待以确保资源释放
                    Thread.sleep(100);
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                }

                // 再次尝试删除文件
                if (file.delete()) {
                    log.info("重试删除文件成功: {}", file.getName());
                } else {
                    log.error("重试删除文件仍然失败: {}", file.getName());
                }
            }
        } else {
            log.warn("要删除的文件不存在: {}", file.getName());
        }

    }
}
